#pragma once

#include <Eigen/Dense>
#include <stdint.h>

namespace  ldl_eigen
{
class OneHot
{
public:
    static Eigen::MatrixXi encode(const Eigen::VectorXi& labels, int num_classes = -1);
    static Eigen::VectorXi decode(const Eigen::MatrixXi& one_hot);
};
}
